python: bouncing ball animation with physics